package com.chaos.auth.service;

import cn.hutool.core.util.ObjectUtil;
import com.chaos.framework.model.dto.ID;
import com.chaos.framework.model.dto.page.PageList;
import com.chaos.framework.model.dto.page.PageQueryDto;
import com.chaos.framework.model.entity.Table;
import com.chaos.framework.model.helper.ConvertHelper;
import com.chaos.framework.model.helper.PageHelper;
import com.chaos.auth.entity.RbacUserRole;
import com.chaos.auth.api.data.RbacUserRoleData;
import com.chaos.auth.mapper.RbacUserRoleMapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

import java.util.List;
import java.util.Set;

/**
 * @author Jacky.Cui
 * @since 2024-11-13
 */
@Slf4j
@Component
public class RbacUserRoleService extends ServiceImpl<RbacUserRoleMapper, RbacUserRole> {

    public ID insertModel(RbacUserRoleData data) {
        RbacUserRole entity = ConvertHelper.convert(data, RbacUserRole.class);
        return save(entity) ? new ID(entity.getId()) : null;
    }

    public boolean deleteModel(ID data) {
        return removeById(data.getId());
    }

    public boolean updateModelByID(RbacUserRoleData data) {
        RbacUserRole entity = ConvertHelper.convert(data, RbacUserRole.class);
        return update(entity, new UpdateWrapper<RbacUserRole>().eq(Table.ID, data.getId()));
    }

    public RbacUserRoleData selectByID(ID data) {
        return ConvertHelper.convert(getOne(new QueryWrapper<RbacUserRole>().eq(Table.ID, data.getId())), RbacUserRoleData.class);
    }

    public List<RbacUserRoleData> selectByData(RbacUserRoleData data) {
        QueryWrapper<RbacUserRole> query = new QueryWrapper<>();
        query.lambda()
                .eq(!ObjectUtil.isEmpty(data.getUserId()), RbacUserRole::getUserId, data.getUserId());
        query.orderByDesc(Table.ID);
        return ConvertHelper.convert(list(query), RbacUserRoleData.class);
    }

    public PageList<RbacUserRoleData> selectByPage(PageQueryDto<RbacUserRoleData> pageData) {
        QueryWrapper<RbacUserRole> query = new QueryWrapper<>();
        query.orderByDesc(Table.ID);
        return PageHelper.pageList(page(PageHelper.page(pageData), query), RbacUserRoleData.class);
    }

    public Set<Long> selectRolesByUserId(Long userId) {
        return baseMapper.selectRolesByUserId(userId);
    }
}
